# -*- coding: utf-8 -*-
"""
author:LTH
data:
"""

from torchvision import datasets, transforms

from config import GetConfig

args = GetConfig()

data_transform = transforms.Compose(
    [
        transforms.Resize([112, 112]),
        transforms.ToTensor(),
        transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
    ]
)

train_dataset = datasets.ImageFolder(root=args.train_path, transform=data_transform)
